/*
 * Copyright 2025 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.beit.mybatis.demo.common.mapper.entity;

import com.google.common.collect.Iterables;
import org.apache.commons.collections4.IterableUtils;
import tk.mybatis.mapper.entity.EntityColumn;
import tk.mybatis.mapper.entity.Example;

import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;

public class CustomOrInCondition {
    private final Example example;

    public CustomOrInCondition(Example example) {
        this.example = example;
    }

    public CustomOrInCondition andIn(Iterable<?> values, String... properties) {
        if (properties == null || IterableUtils.isEmpty(values)) {
            return this;
        }
        String condition = Stream.of(properties).map(example.getPropertyMap()::get).map(EntityColumn::getColumn)
                .collect(Collectors.joining(",", "(", ") in"));
        Example.Criteria criteria = example.and();
        criteria.getCriteria().add(new CustomCriterion(condition, values, Arrays.asList(properties), true));
        return this;
    }

    public CustomOrInCondition andIn(Iterable<?> values, int limitSize, String... properties) {
        Iterable<? extends List<?>> partition = Iterables.partition(values, limitSize);
        String condition = Stream.of(properties).map(example.getPropertyMap()::get).map(EntityColumn::getColumn)
                .collect(Collectors.joining(",", "(", ") in"));
        final List<String> propertiesList = Arrays.asList(properties);
        Example.Criteria criteria = example.and();
        for (List<?> ever : partition) {
            criteria.getCriteria().add(new CustomCriterion(condition, ever, propertiesList,true));
        }
        return this;
    }

    public Example getExample() {
        return example;
    }
}
